package LeetCode.Sort;

import java.util.LinkedList;

public class RadixSort {
    public static void RadixSort(int[] A, int[] temp, int n, int k, int r, int[] cnt) {

        //A:原数组
        //temp:临时数组
        //n:序列的数字个数
        //k:最大的位数2
        //r:基数10
        //cnt:存储bin[i]的个数

        for (int i = 0, rtok = 1; i < k; i++, rtok = rtok * r) {
            //初始化
            for (int j = 0; j < r; j++) {
                cnt[j] = 0;
            }
            //计算每个箱子的数字个数
            for (int j = 0; j < n; j++) {
                cnt[(A[j] / rtok) % r]++;
            }
            //cnt[j]的个数修改为前j个箱子一共有几个数字
            for (int j = 1; j < r; j++) {
                cnt[j] = cnt[j - 1] + cnt[j];
            }
            for (int j = n - 1; j >= 0; j--) {      //重点理解
                cnt[(A[j] / rtok) % r]--;
                temp[cnt[(A[j] / rtok) % r]] = A[j];
            }
            for (int j = 0; j < n; j++) {
                A[j] = temp[j];
            }
        }
    }
}

// 列表法好理解
class Radix {
    public static void main(String[] args) {
        int[] arr = { 23, 1, 4, 9, 98, 132, 42 };
        sort(arr);
    }

    public static void sort(int[] arr) {
        // 1.找 分类-收集 的轮数（最大值的长度）
        int radix = getRadix(arr);
        // 2.创建桶 list所有桶的集合 每一个桶是LinkedList当成队列来用
        LinkedList<Integer>[] list = new LinkedList[10];
        for (int i = 0; i < list.length; i++) {
            list[i] = new LinkedList<>();
        }
        // 3.开始 分类-收集
        for (int r = 1; r <= radix; r++) {
            // 分类过程
            for (int i = 0; i < arr.length; i++) {
                list[getIndex(arr[i], r)].offer(arr[i]);
            }
            int index = 0; // 遍历arr原数组
            // 收集的过程
            for (int i = 0; i < list.length; i++) {
                while (!list[i].isEmpty()) {
                    arr[index++] = list[i].poll();
                }
            }

        }

        for (int i = 0; i < arr.length; i++) {
            System.out.print(arr[i] + "  ");
        }

    }

    private static int getIndex(int num, int r) {
        int ret = 0;
        for (int i = 1; i <= r; i++) {
            ret = num % 10;
            num /= 10;
        }
        return ret;
    }

    private static int getRadix(int[] arr) {
        int max = arr[0];
        for (int i = 1; i < arr.length; i++) {
            if (arr[i] > max) {
                max = arr[i];
            }
        }
        return (max + "").length();
    }
}
